Creating a new matrix from a matrix of index in numpy